package com.jt.base.exception;

import com.jt.base.utils.R;
import org.apache.shiro.authz.UnauthenticatedException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import javax.validation.ConstraintViolation;
import javax.validation.ConstraintViolationException;
import java.util.Set;

/**
 * 处理异常信息
 *
 * @author longheping
 * @date 2018/6/27 17:05
 */
@RestControllerAdvice
public class GlobalExceptionHandler {

  private Logger logger = LoggerFactory.getLogger(getClass());

  /** 处理自定义异常 */
  @ExceptionHandler(RRException.class)
  public R handleRRException(RRException e) {
    R r = new R();
    r.put("code", e.getCode());
    r.put("msg", e.getMessage());
    return r;
  }

  @ExceptionHandler(DuplicateKeyException.class)
  public R handleDuplicateKeyException(DuplicateKeyException e) {
    logger.error(e.getMessage(), e);
    return R.error("数据库中已存在该记录");
  }

  @ExceptionHandler(MethodArgumentNotValidException.class)
  public R handleMethodArgumentNotValidException(MethodArgumentNotValidException e) {
    logger.error(e.getMessage(), e);
    return R.error(e.getBindingResult().getFieldError().getDefaultMessage());
  }

  @ExceptionHandler(ConstraintViolationException.class)
  public R handleConstraintViolationException(ConstraintViolationException e) {
    logger.error(e.getMessage(), e);
    Set<ConstraintViolation<?>> set = e.getConstraintViolations();
    String msg = "";
    int i = 0;
    for (ConstraintViolation<?> constraintViolation : set) {
      msg =
          i == 0
              ? constraintViolation.getMessageTemplate()
              : msg + "," + constraintViolation.getMessageTemplate();
      i++;
    }
    return R.error(msg);
  }

  @ExceptionHandler(UnauthenticatedException.class)
  public R handleUnauthenticatedException(UnauthenticatedException e) {
    logger.error(e.getMessage(), e);
    return R.error("您没有权限操作");
  }

  @ExceptionHandler(Exception.class)
  public R handleException(Exception e) {
    logger.error(e.getMessage(), e);
    return R.error(e.getMessage());
  }
}
